from scipy import zeros,pi,sqrt
from constantes import mJupiter,mSol,G,mTerra
from scipy.linalg import norm

#mapa top left figura 1

dimensao=2
# condicoes inicias massa 1
r1=zeros(dimensao);
m1=mSol;

#condicoes inicias massa 2
#figura 1
periodo=4;
r2=zeros(dimensao);
r2[0]=G*m1*(periodo**2)/(4*pi**2);
r2[0]=r2[0]**(1/3);
m2=mJupiter;
p2=zeros(dimensao);
p2[1]=G*m1/r2[0];
p2[1]=sqrt(p2[1]);
p2=m2*p2;

# condicoes inicias massa 3
a=0.05;
e=0.7;
c=e*a;
b=sqrt(a**2-c**2);
r3=zeros(dimensao);
r3[0]=a*(1-e);
m3=mTerra;
p3=zeros(dimensao);
P=sqrt(4*pi**2*a**3/G*m1);
N=2*pi/P;
p3[1]=a*b*N/r3[0];
temp2=sqrt(G*m1/a);
p3=m3*p3;

# posicao centro massa
rcm=r1*m1+r2*m2+r3*m3;
rcm=rcm/(m1+m2+m3);

# transformacao coordenadas centro massa
r1=r1-rcm;
r2=r2-rcm;
r3=r3-rcm;

#momento linear total nulo
p1=-p2-p3;

H=norm(p1)**(2)/(2*m1)+norm(p2)**(2)/(2*m2)+norm(p3)**(2)/(2*m3)-G*m1*m2/norm(r1-r2)-G*m1*m3/norm(r1-r3)-G*m2*m3/norm(r2-r3);